// Code generated by MockGen. DO NOT EDIT.
// Source: ./router/poolmgr/pool_mgr.go

// Package mock is a generated GoMock package.
package mock

import (
	reflect "reflect"

	gomock "github.com/golang/mock/gomock"
	kr "github.com/pg-sharding/spqr/pkg/models/kr"
	txstatus "github.com/pg-sharding/spqr/pkg/txstatus"
	client "github.com/pg-sharding/spqr/router/client"
	poolmgr "github.com/pg-sharding/spqr/router/poolmgr"
)

// MockConnectionKeeper is a mock of ConnectionKeeper interface.
type MockConnectionKeeper struct {
	ctrl     *gomock.Controller
	recorder *MockConnectionKeeperMockRecorder
}

// MockConnectionKeeperMockRecorder is the mock recorder for MockConnectionKeeper.
type MockConnectionKeeperMockRecorder struct {
	mock *MockConnectionKeeper
}

// NewMockConnectionKeeper creates a new mock instance.
func NewMockConnectionKeeper(ctrl *gomock.Controller) *MockConnectionKeeper {
	mock := &MockConnectionKeeper{ctrl: ctrl}
	mock.recorder = &MockConnectionKeeperMockRecorder{mock}
	return mock
}

// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockConnectionKeeper) EXPECT() *MockConnectionKeeperMockRecorder {
	return m.recorder
}

// ActiveShards mocks base method.
func (m *MockConnectionKeeper) ActiveShards() []kr.ShardKey {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ActiveShards")
	ret0, _ := ret[0].([]kr.ShardKey)
	return ret0
}

// ActiveShards indicates an expected call of ActiveShards.
func (mr *MockConnectionKeeperMockRecorder) ActiveShards() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ActiveShards", reflect.TypeOf((*MockConnectionKeeper)(nil).ActiveShards))
}

// ActiveShardsReset mocks base method.
func (m *MockConnectionKeeper) ActiveShardsReset() {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "ActiveShardsReset")
}

// ActiveShardsReset indicates an expected call of ActiveShardsReset.
func (mr *MockConnectionKeeperMockRecorder) ActiveShardsReset() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ActiveShardsReset", reflect.TypeOf((*MockConnectionKeeper)(nil).ActiveShardsReset))
}

// Client mocks base method.
func (m *MockConnectionKeeper) Client() client.RouterClient {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Client")
	ret0, _ := ret[0].(client.RouterClient)
	return ret0
}

// Client indicates an expected call of Client.
func (mr *MockConnectionKeeperMockRecorder) Client() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Client", reflect.TypeOf((*MockConnectionKeeper)(nil).Client))
}

// SetTxStatus mocks base method.
func (m *MockConnectionKeeper) SetTxStatus(status txstatus.TXStatus) {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "SetTxStatus", status)
}

// SetTxStatus indicates an expected call of SetTxStatus.
func (mr *MockConnectionKeeperMockRecorder) SetTxStatus(status interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTxStatus", reflect.TypeOf((*MockConnectionKeeper)(nil).SetTxStatus), status)
}

// TxStatus mocks base method.
func (m *MockConnectionKeeper) TxStatus() txstatus.TXStatus {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "TxStatus")
	ret0, _ := ret[0].(txstatus.TXStatus)
	return ret0
}

// TxStatus indicates an expected call of TxStatus.
func (mr *MockConnectionKeeperMockRecorder) TxStatus() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TxStatus", reflect.TypeOf((*MockConnectionKeeper)(nil).TxStatus))
}

// MockPoolMgr is a mock of PoolMgr interface.
type MockPoolMgr struct {
	ctrl     *gomock.Controller
	recorder *MockPoolMgrMockRecorder
}

// MockPoolMgrMockRecorder is the mock recorder for MockPoolMgr.
type MockPoolMgrMockRecorder struct {
	mock *MockPoolMgr
}

// NewMockPoolMgr creates a new mock instance.
func NewMockPoolMgr(ctrl *gomock.Controller) *MockPoolMgr {
	mock := &MockPoolMgr{ctrl: ctrl}
	mock.recorder = &MockPoolMgrMockRecorder{mock}
	return mock
}

// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockPoolMgr) EXPECT() *MockPoolMgrMockRecorder {
	return m.recorder
}

// ConnectionActive mocks base method.
func (m *MockPoolMgr) ConnectionActive(rst poolmgr.ConnectionKeeper) bool {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ConnectionActive", rst)
	ret0, _ := ret[0].(bool)
	return ret0
}

// ConnectionActive indicates an expected call of ConnectionActive.
func (mr *MockPoolMgrMockRecorder) ConnectionActive(rst interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConnectionActive", reflect.TypeOf((*MockPoolMgr)(nil).ConnectionActive), rst)
}

// RouteCB mocks base method.
func (m *MockPoolMgr) RouteCB(client client.RouterClient, sh []kr.ShardKey) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "RouteCB", client, sh)
	ret0, _ := ret[0].(error)
	return ret0
}

// RouteCB indicates an expected call of RouteCB.
func (mr *MockPoolMgrMockRecorder) RouteCB(client, sh interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RouteCB", reflect.TypeOf((*MockPoolMgr)(nil).RouteCB), client, sh)
}

// TXBeginCB mocks base method.
func (m *MockPoolMgr) TXBeginCB(rst poolmgr.ConnectionKeeper) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "TXBeginCB", rst)
	ret0, _ := ret[0].(error)
	return ret0
}

// TXBeginCB indicates an expected call of TXBeginCB.
func (mr *MockPoolMgrMockRecorder) TXBeginCB(rst interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TXBeginCB", reflect.TypeOf((*MockPoolMgr)(nil).TXBeginCB), rst)
}

// TXEndCB mocks base method.
func (m *MockPoolMgr) TXEndCB(rst poolmgr.ConnectionKeeper) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "TXEndCB", rst)
	ret0, _ := ret[0].(error)
	return ret0
}

// TXEndCB indicates an expected call of TXEndCB.
func (mr *MockPoolMgrMockRecorder) TXEndCB(rst interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TXEndCB", reflect.TypeOf((*MockPoolMgr)(nil).TXEndCB), rst)
}

// UnRouteCB mocks base method.
func (m *MockPoolMgr) UnRouteCB(client client.RouterClient, sh []kr.ShardKey) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "UnRouteCB", client, sh)
	ret0, _ := ret[0].(error)
	return ret0
}

// UnRouteCB indicates an expected call of UnRouteCB.
func (mr *MockPoolMgrMockRecorder) UnRouteCB(client, sh interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnRouteCB", reflect.TypeOf((*MockPoolMgr)(nil).UnRouteCB), client, sh)
}

// UnRouteWithError mocks base method.
func (m *MockPoolMgr) UnRouteWithError(client client.RouterClient, sh []kr.ShardKey, errmsg error) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "UnRouteWithError", client, sh, errmsg)
	ret0, _ := ret[0].(error)
	return ret0
}

// UnRouteWithError indicates an expected call of UnRouteWithError.
func (mr *MockPoolMgrMockRecorder) UnRouteWithError(client, sh, errmsg interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnRouteWithError", reflect.TypeOf((*MockPoolMgr)(nil).UnRouteWithError), client, sh, errmsg)
}

// ValidateReRoute mocks base method.
func (m *MockPoolMgr) ValidateReRoute(rst poolmgr.ConnectionKeeper) bool {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ValidateReRoute", rst)
	ret0, _ := ret[0].(bool)
	return ret0
}

// ValidateReRoute indicates an expected call of ValidateReRoute.
func (mr *MockPoolMgrMockRecorder) ValidateReRoute(rst interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateReRoute", reflect.TypeOf((*MockPoolMgr)(nil).ValidateReRoute), rst)
}
